from gensim.models import word2vec

def main():
    model = word2vec.Word2Vec.load('./data/w2vec.model')
    req_count = 10
    sbw_list = model.similar_by_word(u'火箭队', topn=100)
    for key in sbw_list:
        if len(key[0]) == 3:
            req_count -= 1
            print(key[0], key[1])
            if req_count == 0:
                break;

if __name__ == '__main__':
    main()